import unittest
from pathlib import Path

from langchain_core.runnables import RunnableConfig
from langgraph.runtime import Runtime

from video_summarize_tools import nodes
from video_summarize_tools.data_models import RuntimeContext, State


class NodeTest(unittest.TestCase):
    def test_video_file_post_process(self):
        file_path = Path("/home/zst/tmp/unite2.mp4")
        context = RuntimeContext()
        context.video_file_path = file_path

        runtime = Runtime(context = context)

        nodes.video_file_post_process(State(), RunnableConfig(),  runtime)
